Why am I getting TypeError: list indices must be integers or slices not float in my Python code